package homework1_array.alg_81;

import static homework1_array.Utils.printList;

public class ArrayHomework {

    public static void main(String[] args) {

        //todo 作业1：完成这个测试方法中的addByElementBySequence()方法
        arrayAddTest();

        // todo 作业2：完成这个测试类中的removeByElement()方法
        deleteTest();

    }

    /**
     * 在有序列表中插入元素
     */
    private static void arrayAddTest() {

        int[] arr = new int[20];
        arr[0] = 3;
        arr[1] = 4;
        arr[2] = 7;
        arr[3] = 8;
// 1.中间位置插入
        addByElementSequence(arr, 4, 6);
        printList("在有序数组的中间部分插入元素后的结果为：", arr, 5);
        //此时应输出 3 4 6 7 8

//2.尾部插入
        addByElementSequence(arr, 5, 9);
        printList("在有序数组的尾部插入元素后的结果为：", arr, 6);
        //此时应输出 3 4 6 7 8 9

//3.数组首部插入
        addByElementSequence(arr, 6, 1);
        printList("在有序数组的首部部分插入元素后的结果为：", arr, 7);
        //此时应输出：1 3 4 6 7 8 9

    }

    //todo 作业1：完成这个方法

    /**
     * @param arr     存储元素的数组
     * @param size    数组中已经存储的元素数量
     * @param element 要插入数组的新元素
     * @return
     */
    public static void addByElementSequence(int[] arr, int size, int element) {
        if( size<0 || size>arr.length){
            throw new IllegalArgumentException("size lt 0 or size gt arr.length");
        }
        //找到插入位置，找到第一个比element大的数字
        int position = size-1;// 默认插入最后
        if(element<arr[0]){
            //首部插入
            position = 0;
        }else if(element>arr[size-1]){
           position = size;
        }else{
            //中间插入
            for (int i = 0; i < size; i++) {
                if(element<arr[i]){
                    position = i;
                    break;
                }
            }
        }
        //从后向前移动元素，将大于position的元素后移，并将element插入到position位置
        for (int i = size-1; i >= position; i--) {
            arr[i+1] = arr[i];
        }
        arr[position] = element;
        size++;
    }

    /**
     * 删除测试
     */
    private static void deleteTest() {
        int[] arr = new int[]{2, 3, 4, 9, 10, 11, 12};

//1.删除头元素
        removeByElement(arr, 7, 2);
        printList("删除头元素后的结果为：", arr, 6);
        //此时应该打印 3 4 9 10 11 12

//2. 删除尾元素
        removeByElement(arr, 6, 12);
        printList("删除尾元素的结果为：", arr, 5);
        //此时应该打印 3 4 9 10 11

//3. 删除中间元素：
        removeByElement(arr, 5, 9);
        printList("删除中间元素的结果为：", arr, 4);
        //此时应该打印 3 4 10 11
    }

    /**
     * 从数组中删除元素e
     *
     * @param arr
     * @param size 数组已经存放的元素大小
     * @param key  要删除的元素
     */
    public static void removeByElement(int[] arr, int size, int key) {
        if( size<0 || size>arr.length){
            throw new IllegalArgumentException("size lt 0 or size gt arr.length");
        }
        // 头部元素
        int position = -1;
        if(key == arr[0]){
            position = 0;
        }else if(key == arr[size-1]){
            //尾部元素
            position = size-1;
        }else{
            //删除中间元素
            //查询删除位置
            for (int i = 1; i < size-1; i++) {
                if(key == arr[i]){
                    position = i;
                }
            }
        }
        if(position != -1){
            //将position之后的数据向前移动
            for (int i = position; i < size-1; i++) {
                arr[i] = arr[i+1];
            }
            size--;
        }
    }
}
